if __name__ == '__main__':
    net = YOLOv1_resnet()
    trainer = torch.optim.SGD(net.parameters(), lr=0.2,  weight_decay=5e-4)
    net.train()
    X = torch.normal(0, 1, (1, 3, 448, 448))
    Y = net(X) #(1,30,7,7)

    print(Y)
    print(Y.size())
    print(Y.max())  
